Skip to content

Building Maxdiffusion images with jax_stable_stack=0.4.35-rev1#126

Merged
parambole merged 1 commit intomainfrom
parambole/jss-trillium
Oct 23, 2024
Merged

Building Maxdiffusion images with jax_stable_stack=0.4.35-rev1#126
parambole merged 1 commit intomainfrom
parambole/jss-trillium

Conversation

@parambole
Copy link
Copy Markdown
Collaborator

We have released a new version of JAX Stable Stack for Trillium i.e. built on top of JAX=0.4.35. Making changes to ensure that Maxdiffusion is built using the latest version so that it can be verified on Trillium.

Tested:

bash .github/workflows/build_and_upload_images.sh CLOUD_IMAGE_NAME=maxdiffusion_jax_stable_stack_0.4.35 MODE=stable_stack PROJECT=tpu-prod-env-multipod LOCAL_IMAGE_NAME=maxdiffusion_jax_stable_stack_0.4.35 BASEIMAGE=us-docker.pkg.dev/cloud-tpu-images/jax-stable-stack/tpu:jax0.4.35-rev1

@parambole parambole force-pushed the parambole/jss-trillium branch from 605bc9d to 22924eb Compare October 23, 2024 17:29
@parambole parambole changed the title Parambole/jss trillium Building Maxdiffusion images with jax_stable_stack=0.4.35-rev1 Oct 23, 2024
@parambole parambole requested a review from entrpn October 23, 2024 17:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants